import json
import os
from datetime import datetime

# 假设我们想要读取的目录路径是 'path_to_directory'
# 这里我们使用一个示例路径，实际的路径需要根据您的需求来确定
directory_path = 'D:\\PythonWorks\\pythonlearning\\01test\\tiantianjijin\\20240520'  # 请确保这个路径存在且包含txt文件

# 列出目录中的所有文件和文件夹
files_in_directory = os.listdir(directory_path)

# 筛选出所有txt文件
txt_files = [file for file in files_in_directory if file.endswith('.txt')]

# 设置当前日期
current_date = datetime.now().strftime('%Y%m%d')
# print(txt_files)
# 假设我们找到了一些txt文件，现在我们将尝试读取并解析这些文件中的JSON内容
json_data = {}

for file in txt_files:
    try:
        # 读取文件内容
        with open(os.path.join(directory_path, file), 'r', encoding='utf-8') as file_obj:
            content = file_obj.read()

        # 解析JSON内容
        parsed_data = json.loads(content)
        # print(parsed_data)

        # 遍历 diff 列表，为每条数据添加 dt 字段
        diff_data = parsed_data.get('data', {}).get('diff', [])
        # for item in data['data']['diff']:
        for item in diff_data:
            item['dt'] = current_date

        # 在最外层添加 trandt 字段
        parsed_data['trandt'] = current_date
        print(parsed_data)


        json_data[file] = parsed_data

    except json.JSONDecodeError:
        # 如果文件内容不是有效的JSON，则跳过该文件
        continue

# json_data
